GlobalJdbcUpdateImpl.java

/**
 * Copyright 2014 Duan Bingnan
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 *   
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.pinus4j.datalayer.update.jdbc;

import java.sql.Connection;
import java.util.ArrayList;
import java.util.List;

import javax.transaction.Transaction;
import javax.transaction.xa.XAResource;

import org.pinus4j.cluster.resources.IDBResource;
import org.pinus4j.datalayer.update.IGlobalUpdate;
import org.pinus4j.entity.meta.EntityPK;
import org.pinus4j.entity.meta.PKValue;
import org.pinus4j.exceptions.DBOperationException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class GlobalJdbcUpdateImpl extends AbstractJdbcUpdate implements IGlobalUpdate {

    public static final Logger LOG = LoggerFactory.getLogger(GlobalJdbcUpdateImpl.class);

    @Override
    public PKValue save(Object entity, String clusterName) {
        List<Object> entities = new ArrayList<Object>(1);
        entities.add(entity);

        PKValue[] pkValues = saveBatch(entities, clusterName, true);

        if (pkValues.length > 0) {
            return pkValues[0];
        }

        return null;
    }

    @Override
    public PKValue[] saveBatch(List<? extends Object> entities, String clusterName, boolean autoGeneratedKeys) {
        Class<?> clazz = entities.get(0).getClass();
        String tableName = entityMetaManager.getTableName(clazz);

        List<PKValue> pks = new ArrayList<PKValue>();

        Transaction tx = null;
        IDBResource dbResource = null;
        try {
            tx = txManager.getTransaction();
            dbResource = this.dbCluster.getMasterGlobalDBResource(clusterName, entityMetaManager.getTableName(clazz));
            Connection conn = dbResource.getConnection();

            int insertCount = 0;
            if (autoGeneratedKeys) {
                pks = _saveBatchWithAutoGeneratedKeys(conn, entities, -1);
                insertCount = pks.size();
            } else {
                insertCount = _saveBatchWithoutAutoGeneratedKeys(conn, entities, -1);
            }

            if (tx != null) {
                tx.enlistResource((XAResource) dbResource);
            } else {
                dbResource.commit();
            }

            if (isCacheAvailable(clazz) && insertCount > 0) {
                primaryCache.incrCountGlobal(clusterName, tableName, insertCount);
            }

            if (isSecondCacheAvailable(clazz) && insertCount > 0) {
                secondCache.removeGlobal(clusterName, tableName);
            }
        } catch (Exception e1) {
            if (tx == null && dbResource != null)
                dbResource.rollback();

            throw new DBOperationException(e1);
        } finally {
            if (tx == null && dbResource != null) {
                dbResource.close();
            }
        }

        return pks.toArray(new PKValue[pks.size()]);
    }

    @Override
    public void update(Object entity, String clusterName) {
        List<Object> entities = new ArrayList<Object>();
        entities.add(entity);
        updateBatch(entities, clusterName);
    }

    @SuppressWarnings({ "rawtypes", "unchecked" })
    @Override
    public void updateBatch(List<? extends Object> entities, String clusterName) {
        Class<?> clazz = entities.get(0).getClass();
        String tableName = entityMetaManager.getTableName(clazz);

        Transaction tx = null;
        IDBResource dbResource = null;
        try {
            tx = txManager.getTransaction();
            dbResource = this.dbCluster.getMasterGlobalDBResource(clusterName, entityMetaManager.getTableName(clazz));

            Connection conn = dbResource.getConnection();

            int updateCount = _updateBatch(conn, entities, -1);

            if (tx != null) {
                tx.enlistResource((XAResource) dbResource);
            } else {
                dbResource.commit();
            }

            // 删除缓存
            if (isCacheAvailable(clazz) && updateCount > 0) {
                List<EntityPK> pks = new ArrayList(entities.size());
                for (Object entity : entities) {
                    pks.add(entityMetaManager.getEntityPK(entity));
                }
                primaryCache.removeGlobal(clusterName, tableName, pks);
            }
            if (isSecondCacheAvailable(clazz) && updateCount > 0) {
                secondCache.removeGlobal(clusterName, tableName);
            }
        } catch (Exception e) {
            if (tx == null && dbResource != null)
                dbResource.rollback();

            throw new DBOperationException(e);
        } finally {
            if (tx == null && dbResource != null) {
                dbResource.close();
            }
        }
    }

    @Override
    public void removeByPk(EntityPK pk, Class<?> clazz, String clusterName) {
        List<EntityPK> pks = new ArrayList<EntityPK>(1);
        pks.add(pk);
        removeByPks(pks, clazz, clusterName);
    }

    @Override
    public void removeByPks(List<EntityPK> pks, Class<?> clazz, String clusterName) {

        Transaction tx = null;
        IDBResource dbResource = null;
        try {
            tx = txManager.getTransaction();
            dbResource = this.dbCluster.getMasterGlobalDBResource(clusterName, entityMetaManager.getTableName(clazz));

            Connection conn = dbResource.getConnection();

            int removeCount = _removeByPks(conn, pks, clazz, -1);

            if (tx != null) {
                tx.enlistResource((XAResource) dbResource);
            } else {
                dbResource.commit();
            }

            // 删除缓存
            String tableName = entityMetaManager.getTableName(clazz);
            if (isCacheAvailable(clazz) && removeCount > 0) {
                primaryCache.removeGlobal(clusterName, tableName, pks);
                primaryCache.decrCountGlobal(clusterName, tableName, removeCount);
            }
            if (isSecondCacheAvailable(clazz) && removeCount > 0) {
                secondCache.removeGlobal(clusterName, tableName);
            }
        } catch (Exception e) {
            if (tx == null && dbResource != null)
                dbResource.rollback();

            throw new DBOperationException(e);
        } finally {
            if (tx == null && dbResource != null) {
                dbResource.close();
            }
        }
    }

}